import random
from copy import deepcopy

from analysis import generator

data1 = generator.generateRandomInt(50, 70)


def temp_data3():
    equipment = generator.generateRandomTextList("equipment_", repeat=False)  # 默认生成10-20个
    prop = generator.generateRandomTextList("prop_", repeat=False)  # 默认生成10-20个
    material = generator.generateRandomTextList("material_", repeat=False)  # 默认生成10-20个
    items = generator.generateTextListFromList(equipment + prop + material,  # 截取20-60个
                                               amount=random.randint(20, len(equipment + prop + material)))
    return equipment, prop, material, items


data3 = temp_data3()


def temp_data4():
    item_values = generator.generateRandomTextList("item_", repeat=False)  # 默认生成10-20个
    items = generator.generateTextListFromList(item_values, amount=random.randint(7, len(item_values)))  # 截取7-20个
    return {item: generator.generateRandomInt(50, 500) for item in item_values}, items


data4 = temp_data4()
data5 = generator.generateMatrix()


def generate_data1() -> int:
    return deepcopy(data1)


def generate_data2():
    return


def generate_data3():
    return deepcopy(data3)


def generate_data4():
    return deepcopy(data4)


def generate_data5():
    return deepcopy(data5)


print(generator.generateMatrix(5, 4))
